Create 2d array in Python using dict